package com.skh.binaryTree;

import java.util.LinkedList;

/**
 * @Author: skh
 * @Date: 2020/3/3 14:56
 * @Description: 翻转二叉树
 */
public class InvertTree {
    /**
     * 翻转一棵二叉树。
     *
     * 示例：
     *
     * 输入：
     *
     *      4
     *    /   \
     *   2     7
     *  / \   / \
     * 1   3 6   9
     * 输出：
     *
     *      4
     *    /   \
     *   7     2
     *  / \   / \
     * 9   6 3   1
     *
     */

    /**
     * 递归
     * @param root
     * @return
     */
    public TreeNode invertTree(TreeNode root) {
        if (root == null) {
            return null;
        }

        TreeNode right = invertTree(root.right);
        TreeNode left = invertTree(root.left);

        root.left = right;
        root.right = left;

        return root;
    }

    /**
     * 递归
     * @param root
     * @return
     */
    public TreeNode invertTree1(TreeNode root) {
        if (root == null) {
            return null;
        }

        //交换两个节点
        TreeNode temp = root.left;
        root.left = root.right;
        root.right = temp;

        //利用递归翻转子节点
        invertTree1(root.left);
        invertTree1(root.right);

        return root;
    }

    /**
     * 这个方法的思路就是，我们需要交换树中所有节点的左孩子和右孩子。
     * 因此可以创一个队列来存储所有左孩子和右孩子还没有被交换过的节点。
     * 开始的时候，只有根节点在这个队列里面。
     * 只要这个队列不空，就一直从队列中出队节点，然后互换这个节点的左右孩子节点，接着再把孩子节点入队到队列，
     * 对于其中的空节点不需要加入队列。最终队列一定会空，
     * 这时候所有节点的孩子节点都被互换过了，直接返回最初的根节点就可以了。
     *
     * @param root
     * @return
     */
    public TreeNode invertTree2(TreeNode root) {
        if (root == null) {
            return null;
        }

        LinkedList<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        while (!queue.isEmpty()) {
            TreeNode node = queue.remove();



            //交换这个节点的左右两个节点
            TreeNode temp = node.left;
            node.left = node.right;
            node.right = temp;

            if (node.left != null) {
                queue.add(node.left);
            }
            if (node.right != null) {
                queue.add(node.right);
            }
        }


        return root;
    }
}
